package fiuba.algo3.vista;

import java.util.HashMap;
import java.util.Map;

import fiuba.algo3.modelo.vehiculo.*;
import fiuba.algo3.modelo.factorVial.*;

@SuppressWarnings("rawtypes")
public class Imagenes {

	private Map<Class, String> diccionario;
	
	
	public Imagenes(){
		diccionario = new HashMap<Class, String>();
		diccionario.put(Auto.class, "auto.png");
		diccionario.put(TodoTerreno.class, "todoTerreno.png");
		diccionario.put(Moto.class, "moto.png");
		diccionario.put(Pozo.class, "pozo.png");
		diccionario.put(ControlPolicial.class, "controlPolicial.png");
		diccionario.put(Piquete.class, "piquete.png");
		diccionario.put(SorpresaCambio.class, "sorpresa.png");
		diccionario.put(SorpresaDesfavorable.class, "sorpresa.png");
		diccionario.put(SorpresaFavorable.class, "sorpresa.png");
	}
	
	
	public String getImagenDe(Class clase){
		if (diccionario.containsKey(clase))	
			return diccionario.get(clase);
		else return null;
	}

}
